from .llm_openai import LLMOpenAI

SUPPORTED_LLM_AGREEMENTS = {
    "openai": LLMOpenAI
}

def get_llm(llm_name_agreement="openai", api_key=None, base_url=None, model=None):
    return SUPPORTED_LLM_AGREEMENTS[llm_name_agreement](model, api_key, base_url)